import grok
from BTrees.OIBTree import OIBTree

class ContaVotos(grok.Application, grok.Container):
    def __init__(self):
        super(ContaVotos,self).__init__()
        self.votos = OIBTree()
        
    def votoPara(self, nome):
        if nome in self.votos:
            self.votos[nome] += 1
        else:
            self.votos[nome] = 1

class Index(grok.View):
    def update(self, nome=None):
        if nome is not None:
            self.context.votoPara(nome)
            
    def listarVotos(self):
        votos = self.context.votos
        return [{'nome':chave,'votos':votos[chave]} for chave in votos]
        
    
